import pymysql as sql
import pandas as pd
#from alg import stock_info
#识别二买，并截图。

def get_idlist():
    db = sql.connect(host='localhost', port=3306, user='root', password="254001fangfeifei")
    cur = db.cursor()
    cur.execute('use stock')
    cur.execute('select id from idlist')
    data = cur.fetchall()
    #[[id,name,status],...]
    all_id = [i[0] for i in data]
    return all_id

def get_idlist2():
    data = pd.read_csv('d://all_stock.csv')
    l = [i.iloc[1] for i in data]
    print(l)

if __name__ == "__main__":
    #start 300
    get_idlist2()


